from transformer_nuggets.fp8 import scaled_quant